CUDA_VISIBLE_DEVICES=0 python -m torch.distributed.launch --standalone --nproc_per_node=1 --master_port 2023 train.py